from app.models.database import db
from datetime import datetime
import json

class Environment(db.Model):
    """环境配置模型"""
    __tablename__ = 'environment'
    
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)
    description = db.Column(db.Text)
    base_url = db.Column(db.String(255))
    variables = db.Column(db.JSON)  # 环境变量
    headers = db.Column(db.JSON)  # 全局请求头
    config = db.Column(db.Text)  # 环境配置，JSON格式
    auth_config = db.Column(db.Text)  # 认证配置，JSON格式
    created_at = db.Column(db.DateTime, default=datetime.now)
    updated_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now)
    
    # 关联到执行记录
    records = db.relationship('ExecutionRecord', back_populates='environment')
    
    def to_dict(self):
        # 合并两个模型的数据结构，同时支持variables/headers和config/auth_config
        return {
            'id': self.id,
            'name': self.name,
            'description': self.description,
            'base_url': self.base_url,
            'variables': self.variables or {},
            'headers': self.headers or {},
            'config': json.loads(self.config) if self.config else {},
            'auth_config': json.loads(self.auth_config) if self.auth_config else {},
            'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S'),
            'updated_at': self.updated_at.strftime('%Y-%m-%d %H:%M:%S')
        }